#!/usr/bin/env python
#-*-coding:utf-8-*-


from predict import  Predictor
import conf
predictor = Predictor(map_file=conf.TAG_MAP_PATH, checkpoint_dir=conf.TAG_CHECKPOINT_DIR)


def tag_by_local_model(words):
    tags = predictor.predict(words)
    return tags


if __name__ == '__main__':
    words = ['岗位', '职责', '：', '1', '.', '需求', '设计', '与', '开发', '工作', '2', '.', '代码', '规范', '、', '优化', '3',
             '.', '细致', '认真', '，', '善于', '沟通', '，', '理解', '学习', '能力', '强', '任职', '资格', '：', '1', '.',
             'Java', '基础', '扎实', '；', '具有', '三', '年', '以上', 'Java', '开发', '经验', '；', '2', '.', '精通', '后',
             '端', '开发', '相关', '技术', '，', '熟悉', 'springmvc', '、', 'mybatis', '；', '3', '.', '熟悉', 'javaee',
             '规范', '，', '熟悉', '常用', '设计', '模式', '4', '.', '熟悉', 'Linux', '平台', '常用', '命令', '5', '.', '熟悉',
             '数据库', '技术', '，', 'Sql', '优化', '6', '.', '熟悉', '大', '流量', '，', '高', '并发', '，', '高性能', '的',
             '分布式', '系统', '设计', '，', '擅长', '性能', '调', '优', '7', '.', '擅长', '系统', '分析', '设计', '，', '有',
             '大型', '项目', '基础', '与', '应用', '架构', '设计', '经验', '者', '优先']

    tag = tag_by_local_model(words)

    print(len(tag))
    print(len(words))
    print(tag)
